import torch
from torchvision import models
from model import MobileNetV3

if __name__ == '__main__':
    model = MobileNetV3()
    inputs = torch.ones((1, 3, 224, 224))
    output = model(inputs)
